package com.xxgc.helloworld.config;

import com.fasterxml.jackson.annotation.JsonAutoDetect;
import com.fasterxml.jackson.annotation.PropertyAccessor;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cache.CacheManager;
import org.springframework.cache.annotation.CachingConfigurerSupport;
import org.springframework.cache.annotation.EnableCaching;
import org.springframework.cache.interceptor.KeyGenerator;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.cache.RedisCacheConfiguration;
import org.springframework.data.redis.cache.RedisCacheManager;
import org.springframework.data.redis.cache.RedisCacheWriter;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;

import java.lang.reflect.Method;
import java.time.Duration;

@Configuration
@EnableCaching
public class RedisConfig extends CachingConfigurerSupport {
        @Value("${spring.redis.host}")
        private String host;
        @Value("${spring.redis.port}")
        private int port;
        @Value("${spring.redis.timeout}")
        private int timeout;
        @Value("${spring.redis.password}")
        private String password;
        @Value("${spring.redis.jedis.pool.max-active}")
        private int maxActive;
        @Value("${spring.redis.jedis.pool.max-wait}")
        private int maxWait;
        @Value("${spring.redis.jedis.pool.max-idle}")
        private int maxIdle;
        @Value("${spring.redis.jedis.pool.min-idle}")
        private int minIdle;

        @Bean
        public KeyGenerator wiselyKeyGenerator(){
                return new KeyGenerator() {
                        @Override
                        public Object generate(Object target, Method method, Object... params) {
                                StringBuilder sb = new StringBuilder();
                                sb.append(target.getClass().getName());
                                sb.append(method.getName());
                                for (Object obj : params) {
                                        sb.append(obj.toString());
                                }
                                return sb.toString();
                        }
                };
        }

        @Bean
        public JedisConnectionFactory redisConnectionFactory() {
                JedisConnectionFactory factory = new JedisConnectionFactory();
                factory.setHostName(host);
                factory.setPort(port);
                factory.setTimeout(timeout); //设置连接超时时间
                factory.setPassword(password);
                factory.getPoolConfig().setMaxIdle(maxIdle);
                factory.getPoolConfig().setMinIdle(minIdle);
                factory.getPoolConfig().setMaxTotal(maxActive);
                factory.getPoolConfig().setMaxWaitMillis(maxWait);
                return factory;
        }

        // 管理缓存
        @Bean
        public CacheManager cacheManager(RedisConnectionFactory redisConnectionFactory) {
                RedisCacheConfiguration redisCacheConfiguration = RedisCacheConfiguration.defaultCacheConfig()
                        .entryTtl( Duration.ofHours(1)); // 设置缓存有效期一小时
                return RedisCacheManager
                        .builder( RedisCacheWriter.nonLockingRedisCacheWriter(redisConnectionFactory))
                        .cacheDefaults(redisCacheConfiguration).build();
        }

        @Bean
        public RedisTemplate<String, String> redisTemplate(RedisConnectionFactory factory) {
                StringRedisTemplate template = new StringRedisTemplate(factory);
                setSerializer(template); //设置序列化工具，这样ReportBean不需要实现Serializable接口
                template.afterPropertiesSet();
                return template;
        }

        private void setSerializer(StringRedisTemplate template) {
                Jackson2JsonRedisSerializer jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer(Object.class);
                ObjectMapper om = new ObjectMapper();
                om.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
                om.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
                jackson2JsonRedisSerializer.setObjectMapper(om);
                template.setValueSerializer(jackson2JsonRedisSerializer);
        }

}